#pragma once

#include <iostream>
#include <string>
#include <queue>
#include <pthread.h>

const static int defaultcap = 5;

template<typename T>
class BlockQueue
{
private:
    bool IsFull()
    {
        return _block_queue.size() == _max_cap;
    }
    bool IsEmpty()
    {
        return _block_queue.empty();
    }
public:
    BlockQueue(int cap = defaultcap) : _max_cap(cap), _low_water(cap / 3), _high_water(cap / 3 * 2)
    {
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_p_cond, nullptr);
        pthread_cond_init(&_c_cond, nullptr);
    }
    // 假设：两个消费者
    void Pop(T* out)
    {
        pthread_mutex_lock(&_mutex);
        while (IsEmpty()) // if ?，while可以保证代码的鲁棒性（健壮性）
        {
            // 添加尚不满足，但是线程被异常唤醒的情况，叫做伪唤醒
            pthread_cond_wait(&_c_cond, &_mutex); // 两个消费者都在这里等
        }
        // 1.阻塞队列没空 || 2.被唤醒了
        *out = _block_queue.front();
        _block_queue.pop();
        pthread_mutex_unlock(&_mutex);
        // if (_block_queue.size() < _high_water)
            pthread_cond_signal(&_p_cond);
    }
    // 假设：只有一个生产者
    void Equeue(const T& in)
    {
        pthread_mutex_lock(&_mutex);
        while (IsFull()) // if ?
        {
            // 满了，生产者不能生产，必须等待
            // 可是在临界区呀
            // pthread_cond_wait被调用的时候，除了将自己排队等待，还会自己原子性的释放传入的锁
            // 函数返回的时候，不是还在临界区吗？
            // 返回时：必须先参与锁的竞争，重新加上锁，该函数才会返回
            pthread_cond_wait(&_p_cond, &_mutex);

        }
        // 1.阻塞队列没满，|| 2.被唤醒了
        _block_queue.push(in);// 生产到阻塞队列
        pthread_mutex_unlock(&_mutex);
        // 让消费者消费
        // if (_block_queue.size() > _low_water)
            pthread_cond_signal(&_c_cond); //
    }
    ~BlockQueue()
    {
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_p_cond);
        pthread_cond_destroy(&_c_cond);
    }
private:
    int _max_cap;
    pthread_mutex_t _mutex;
    pthread_cond_t _p_cond; // 生产者条件变量
    pthread_cond_t _c_cond; // 消费者条件变量
    int _low_water;
    int _high_water;
    std::queue<T> _block_queue; //临界资源
};